// 动态规划 - 核心 5 步：
// 1. 确定状态表示 - 根据 题目要求，经验(以 i,j 位置为结尾/开始......)，发现重复子问题 确定状态表示
// 2. 推导状态转移方程: dp[i] = ?
//    用 之前的状态 或者 之后的状态 推导当前的状态（根据最近一步划分问题）
// 3. 初始化：保证填表时不越界，结合多开数组的技巧
// 4. 确定填表顺序：填写当前状态值的时候，所需状态的值已经计算过了
// 5. 返回值：结合题目要求 + 状态表示

// 经典题目：斐波那契数列模型，路径问题，简单多状态，子数组

// 技巧：
// dp[] 表多开一个长度，处理数组越界及初始化复杂的问题
// dp[][] 表多开一行，多开一列
// 结合滚动数组优化 - 注意赋值顺序

// 总结经验:
// 动态规划题目如果定义完 dp[] 数组，发现 dp[i] 依赖前面的状态，也依赖后面的状态，那么想一想打家劫舍模型
// 如果觉得不像打家劫舍模型，那么搞一个数组预处理一下，搞成连续的数组，往打家劫舍模型上靠
// 如果题目的状态表示存在多个状态，比如给房子涂颜色（红蓝绿），某个位置元素（选或不选），
// 可以根据经验(以某个位置为结尾/开头)以及状态（定义多个状态: f[i], g[i]）定义状态表示
// 如果动态规划过程中涉及到状态转换，需要画状态机图进行分析
// 如果是环形数组，或者使用分类讨论的方法，或者用“正难则反”的思路，转换为普通数组问题

// 例题 5:
// 如果一个数列 至少有三个元素 ，并且任意两个相邻元素之差相同，则称该数列为等差数列。
//
//        例如，[1,3,5,7,9]、[7,7,7,7] 和 [3,-1,-5,-9] 都是等差数列。
//        给你一个整数数组 nums ，返回数组 nums 中所有为等差数组的 子数组 个数。
//
//        子数组 是数组中的一个连续序列。
//
//        示例 1：
//
//        输入：nums = [1,2,3,4]
//        输出：3
//        解释：nums 中有三个子等差数组：[1, 2, 3]、[2, 3, 4] 和 [1,2,3,4] 自身。
//        示例 2：
//
//        输入：nums = [1]
//        输出：0
//
//
//        提示：
//
//        1 <= nums.length <= 5000
//        -1000 <= nums[i] <= 1000

// 解题思路:
// dp[i] 为以 i 位置为结尾的的等差数列的个数
// if(nums[i] - nums[i - 1] == nums[i - 1] - nums[i - 2]):
// dp[i] = dp[i - 1] + 1
// else: dp[i] = 0

public class NumberOfArithmeticSlices {
    public int numberOfArithmeticSlices(int[] nums) {
        int n = nums.length;
        int[] dp = new int[n];

        int ret = 0;
        for(int i = 2; i < n; i++){
            if(nums[i] - nums[i - 1] == nums[i - 1] - nums[i - 2]){
                dp[i] = dp[i - 1] + 1;
            }else{
                dp[i] = 0;
            }

            ret += dp[i];
        }
        return ret;
    }
}
